package vote.ai.domain.enums;

import lombok.Getter;

@Getter
public enum CorrectionModel {

    /**
     * 调用讯飞API接口进行文本纠错
     */
    XUNFEI_API("讯飞API"),
    /**
     * 使用对话百度AI 进行纠错
     */
    BAIDU_AI("BAIDU_AI"),
    /**
     * gradio_client
     */
    GRADIO_AI("GRADIO_AI"),
    /**
     * 扣子
     */
    COZE("COZE"),
    /**
     * 爱校对
     */
    I_JIAO_DUI("I_JIAO_DUI"),
    /**
     * 本地库匹配
     */
    LOCAL("LOCAL"),
    /**
     * deepseek
     */
    DEEPSEEK("DEEPSEEK"),
    /**
     * deepseek 官方
     */
    DEEPSEEK_GF("DEEPSEEK_GF"),
    /**
     * deepseek app
     */
    DEEPSEEK_APP("DEEPSEEK_APP"),
    /**
     * 通义千问
     */
    TONG_YI("TONG_YI"),
    /**
     * 星火大模型
     */
    HUO_SHAN("HUO_SHAN"),
    /**
     * 腾讯混元
     */
    HUN_YUAN("HUN_YUAN"),
    /**
     * 自定义模型-训练
     */
    CUSTOM_MODEL("自定义模型");

    private final String desc;

    CorrectionModel(String desc) {
        this.desc = desc;
    }

    /**
     * 获取对应服务的bean名称
     *
     * @return
     */
    public String getBeanName() {
        String nameWithoutUnderscore = name().replace("_", "").toLowerCase();
        return nameWithoutUnderscore + "CorrectionService";
    }
}
